package circlequeue

import (
	"errors"
	"fmt"
	"os"
)

// 定义环形数组结构体
type CircleQueue struct {
	maxSize int    // 默认5
	array   [5]int // 默认数组中有5个
	head    int    // 指向队列队首，初始值为0
	tail    int    // 指向队列队尾，初始值为0
}

// 往队列中添加值
func (this *CircleQueue) Push(val int) (err error) {
	if this.IsFull() {
		err = errors.New("队列满了")
		return
	}

	this.array[this.tail] = val
	this.tail = (this.tail + 1) % this.maxSize

	return
}

// 从队列中取值
func (this *CircleQueue) Pop() (val int, err error) {
	if this.IsEmpty() {
		err = errors.New("队列中没有数据")
		return
	}

	val = this.array[this.head]
	this.head = (this.head + 1) % this.maxSize

	return
}

// 显示队列
func (this *CircleQueue) ListQueue() {
	fmt.Println("环形队列情况如下：")
	// 取出当前队列有多少个元素
	size := this.Size()
	if size == 0 {
		fmt.Println("队列为空")
	}

	tempHead := this.head
	for i := 0; i < size; i++ {
		fmt.Printf("array[%d]=%d\t", tempHead, this.array[tempHead])
		tempHead = (tempHead + 1) % this.maxSize
	}
	fmt.Println()
}

// 队列是否满了
func (this *CircleQueue) IsFull() bool {
	return (this.tail+1)%this.maxSize == this.head
}

// 队列是否为空
func (this *CircleQueue) IsEmpty() bool {
	return this.tail == this.head
}

// 取出环形队列有多少个元素
func (this *CircleQueue) Size() int {
	// 这是一个关键算法
	return (this.tail + this.maxSize - this.head) % this.maxSize
}

func TestCircleQueue() {
	queue := &CircleQueue{
		maxSize: 5,
		head:    0,
		tail:    0,
	}

	var key string
	var val int
	for {
		fmt.Println("1.输入 add 表示添加数据到队列")
		fmt.Println("2.输入 get 表示从队列获取数据")
		fmt.Println("3.输入 show 表示显示队列")
		fmt.Println("4.输入 exit 表示退出")

		fmt.Scanln(&key)

		switch key {
		case "add":
			fmt.Println("输入你要入队的值")
			fmt.Scanln(&val)
			err := queue.Push(val)
			if err != nil {
				fmt.Println(err.Error())
			} else {
				fmt.Println("加入队列ok")
			}
		case "get":
			val, err := queue.Pop()
			if err != nil {
				fmt.Println(err.Error())
			} else {
				fmt.Printf("取出的值是：%d\n", val)
			}
		case "show":
			queue.ListQueue()
		case "exit":
			os.Exit(0)
		}
	}
}
